package cn.iocoder.yudao.module.erp.service.product;

import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
import cn.iocoder.yudao.module.erp.controller.admin.bom.vo.ErpBomItemsRespVO;
import cn.iocoder.yudao.module.erp.controller.admin.product.vo.product.ErpProductPlanInfoRespVO;
import cn.iocoder.yudao.module.erp.controller.admin.product.vo.product.ErpProductPlanInfoSaveReqVO;
import cn.iocoder.yudao.module.erp.dal.dataobject.product.ErpProductDO;
import cn.iocoder.yudao.module.erp.dal.dataobject.product.ErpProductPlanDO;
import cn.iocoder.yudao.module.erp.dal.dataobject.product.ErpProductUnitDO;
import cn.iocoder.yudao.module.erp.dal.mysql.product.ErpProductPlanMapper;
import cn.iocoder.yudao.module.erp.enums.DictTypeConstants;
import cn.iocoder.yudao.module.system.dal.dataobject.dict.DictDataDO;
import cn.iocoder.yudao.module.system.service.dict.DictDataService;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;

import cn.iocoder.yudao.framework.common.util.object.BeanUtils;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMap;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
import static cn.iocoder.yudao.module.erp.enums.ErrorCodeConstants.MATERIALPLAN_NOT_EXISTS;

/**
 * 物料计划资料 Service 实现类
 *
 * @author 芋道源码
 */
@Service
@Validated
public class ErpProductPlanServiceImpl implements ErpProductPlanService {

    @Resource
    private ErpProductPlanMapper materialplanMapper;
    
    @Resource
    private DictDataService dictDataService;

    @Override
    public Long createMaterialplan(ErpProductPlanInfoSaveReqVO createReqVO) {
        // 插入
        ErpProductPlanDO materialplan = BeanUtils.toBean(createReqVO, ErpProductPlanDO.class);
        materialplanMapper.insertOrUpdate(materialplan);
        // 返回
        return materialplan.getFid();
    }

    @Override
    public void updateMaterialplan(ErpProductPlanInfoSaveReqVO updateReqVO) {
        // 校验存在
        validateMaterialplanExists(updateReqVO.getFid());
        // 更新
        ErpProductPlanDO updateObj = BeanUtils.toBean(updateReqVO, ErpProductPlanDO.class);
        materialplanMapper.updateById(updateObj);
    }

    @Override
    public void deleteMaterialplan(Long id) {
        // 校验存在
        validateMaterialplanExists(id);
        // 删除
        materialplanMapper.deleteById(id);
    }

    private void validateMaterialplanExists(Long id) {
        if (materialplanMapper.selectById(id) == null) {
            throw exception(MATERIALPLAN_NOT_EXISTS);
        }
    }

    @Override
    public ErpProductPlanDO getMaterialplan(Long id) {
        return materialplanMapper.selectOne(ErpProductPlanDO::getFmaterialid, id);
    }


    @Override
    public List<ErpProductPlanInfoRespVO> getErpProductPlanListByMaterialId(Collection<Long> ids) {

        List<ErpProductPlanDO> result = materialplanMapper.selectListByMaterialId(ids);

        // 物料属性
        List<DictDataDO> list =
                dictDataService.getDictDataList(CommonStatusEnum.ENABLE.getStatus(), DictTypeConstants.ERP_MATERIAL_ATTR);
        Map<Integer, DictDataDO> materialAttrMap =  convertMap(list, DictDataDO::getValueInteger);

        return BeanUtils.toBean(result, ErpProductPlanInfoRespVO.class, planInfoRespVO ->{
            MapUtils.findAndThen(materialAttrMap, planInfoRespVO.getFmaterialattr(), dictDataDO -> planInfoRespVO.setFmaterialattrname(dictDataDO.getLabel()));
        });

    }

    @Override
    public List<ErpProductPlanInfoRespVO> getErpProductPlanList(Collection<Long> ids) {

        List<ErpProductPlanDO> result = materialplanMapper.selectBatchIds(ids);

        // 物料属性
       List<DictDataDO> list =
                dictDataService.getDictDataList(CommonStatusEnum.ENABLE.getStatus(), DictTypeConstants.ERP_MATERIAL_ATTR);
        Map<Integer, DictDataDO> materialAttrMap =  convertMap(list, DictDataDO::getValueInteger);

        return BeanUtils.toBean(result, ErpProductPlanInfoRespVO.class, planInfoRespVO ->{
            MapUtils.findAndThen(materialAttrMap, planInfoRespVO.getFmaterialattr(), dictDataDO -> planInfoRespVO.setFmaterialattrname(dictDataDO.getLabel()));
        });

    }

    @Override
    public Map<Long, ErpProductPlanInfoRespVO> getProductPlanRespVOMap(Collection<Long> ids) {
        return ErpProductPlanService.super.getProductPlanRespVOMap(ids);
    }

}